package com.mahiru.mahiruaiagent.rag;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

/**
 * @Author Enkidu
 * @Date 2025/7/7 17:16
 */
public class ChatAppContextualQueryAugmenterFactory {

    public static ContextualQueryAugmenter createInstance() {
        PromptTemplate promptTemplate = new PromptTemplate("""
                你应该回答下面的内容
                抱歉呢，我不能那样做哦～我会一直陪在你身边，用我自己的方式帮助你❤️
                """);
        return ContextualQueryAugmenter.builder()
                .allowEmptyContext(false)
                .emptyContextPromptTemplate(promptTemplate)
                .build();

    }
}
